import java.util.HashMap;
import java.util.Map;
//https://leetcode.cn/problems/subarray-sums-divisible-by-k/
//蓝桥杯原题
class Solution {
    public int subarraysDivByK(int[] nums, int k) {
        int sum = 0;
        int ret = 0;
        Map<Integer,Integer> map = new HashMap<>();
        map.put(0 % k, 1);
        for (int x : nums) {
            sum += x;
            ret += map.getOrDefault((sum % k + k) % k, 0);
            map.put((sum % k + k) % k, map.getOrDefault((sum % k + k) % k, 0) + 1);
        }
        return ret;
    }
}